import cv2

# RTSP 地址
rtsp_url = "rtsp://chi-square.com.cn:6004/live/test"

# 打开 RTSP 流
cap = cv2.VideoCapture(rtsp_url)

if not cap.isOpened():
    print("无法打开 RTSP 流")
    exit()

while True:
    # 读取一帧图像
    ret, frame = cap.read()

    if not ret:
        print("无法读取帧")
        break

    # 在这里可以对帧进行处理，比如显示、保存等
    cv2.imshow('RTSP Stream', frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()